'''
第一周作业，定义一个统计指定目录大小的函数
'''
import os

def DirSize(d):
    sum = 0
    # 判断指定目录是否为文件
    if os.path.isfile(d):
        sum += os.path.getsize(d)
    # 判断指定目录是否为文件夹
    if os.path.isdir(d):
        DirList = os.listdir(d)
        for f in DirList:
            file = os.path.join(d,f)
            if os.path.isfile(file):
                sum += os.path.getsize(file)
            if os.path.isdir(file):
                sum += DirSize(file)
    return sum

while True:
    key = input("请输入指定文件目录的文件夹或文件：")
    if key == "":
        break
    else:
        print("{}字节".format(DirSize(key)))